# -*- coding: utf-8 -*-
"""
@Env 
@Time 2024/9/3 下午2:23
@Author yzpang
@Function: 模型相关配置
"""
import os

SYSTEM = os.name

# 模型文件绝对路径
if SYSTEM == 'nt':
    MODEL_ROOT_PATH = r"E:\AI\models"
    DATASET_ROOT_PATH = r"E:\AI\datasets"
else:
    MODEL_ROOT_PATH = r"/home/models"
    DATASET_ROOT_PATH = r"/home/datasets"

# 返回识别结果数量
TOP_K = 5

# 学习率
DEFAULT_LR = 5e-5
# 批次大小
DEFAULT_BATCH_SIZE = 8
# 训练轮数
DEFAULT_EPOCHS = 5
# 训练设备
DEFAULT_DEVICE = 'cpu'


